package cn.sherer.utils;

import java.net.URL;
import java.io.File;
import java.util.Map;
import java.util.HashMap;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLConnection;
import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.util.logging.Level;
import java.io.DataInputStream;
import java.io.FileInputStream;
import java.io.DataOutputStream;
import java.util.logging.Logger;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;

/**
 * 网络工具类
 */
public class WebUtils {

    /**
     * 请求方法
     */
    public enum Method { GET, POST, PUT, DELETE }
    /**
     * 通用 User-Agent
     */
    public static final String GENERAL_AGENT = "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1;SV1)";
    /**
     * 微信 User-Agent
     */
    public static final String WECHAT_AGENT = "Mozilla/5.0 (Linux; Android 6.0; DIG-AL00 Build/HUAWEIDIG-AL00; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/53.0.2785.49 Mobile MQQBrowser/6.2 TBS/043124 Safari/537.36 MicroMessenger/6.5.6.1020 NetType/4G Language/zh_CN";
    /**
     * 抖音 User-Agent
     */
    public static final String DOUYIN_AGENT = "Mozilla/5.0 (Linux; Android 12; M2007J17C Build/SKQ1.211006.001; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/88.0.4324.181 Mobile Safari/537.36  aweme_240700 JsSdk/1.0 NetType/WIFI Channel/xiaomi_1128_64 AppName/aweme app_version/24.7.0 ByteLocale/zh-CN Region/CN AppSkin/black AppTheme/dark  BULLET/1 BytedanceWebview/d8a21c6 TTWebView/0881130046418";
    /**
     * 快手 User-Agent
     */
    public static final String KUAISHOU_AGENT = "Mozilla/5.0 (Linux; Android 12; M2007J17C Build/SKQ1.211006.001; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/90.0.4430.226 KsWebView/1.8.90.603 (rel) Mobile Safari/537.36 NetType/wifi Language/zh_CN miniProgram/1.57.2 KUAISHOU/11.3.40.30709 aegon/3.30.1";


    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @return URL所代表远程资源的响应结果
     */
    public static String get(String url) {
        return get(url, "");
    }

    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static String get(String url, String userAgent) {
        return get(url, userAgent, null);
    }

    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static String get(String url, String userAgent, Map<String, String> headers) {
        return get(url, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static String get(String url, String userAgent, Map<String, String> headers, int timeout) {
        return request(url, Method.GET, "", userAgent, headers, 0);
    }


    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @return URL所代表远程资源的响应结果
     */
    public static String post(String url) {
        return post(url, "");
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @return URL所代表远程资源的响应结果
     */
    public static String post(String url, String parameter) {
        return post(url, parameter, "");
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static String post(String url, String parameter, String userAgent) {
        return post(url, parameter, userAgent, null);
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static String post(String url, String parameter, String userAgent, Map<String, String> headers) {
        return post(url, parameter, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static String post(String url, String parameter, String userAgent, Map<String, String> headers, int timeout) {
        return request(url, Method.POST, parameter, userAgent, headers, 0);
    }


    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url) {
        return request(url, Method.GET, "");
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, Method method, String parameter) {
        return request(url, method, parameter, "");
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, Method method, String parameter, String userAgent) {
        return request(url, method, parameter, userAgent, null);
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, Method method, String parameter, String userAgent, Map<String, String> headers) {
        return request(url, method, parameter, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, Method method, String parameter, String userAgent, Map<String, String> headers, int timeout) {
        return getText(requestByte(url, method, parameter, userAgent, headers, timeout));
    }


    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] getByte(String url) {
        return getByte(url, "");
    }

    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] getByte(String url, String userAgent) {
        return getByte(url, userAgent, null);
    }

    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] getByte(String url, String userAgent, Map<String, String> headers) {
        return getByte(url, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送GET请求
     * @param url 发送请求的URL
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] getByte(String url, String userAgent, Map<String, String> headers, int timeout) {
        return requestByte(url, Method.GET, "", userAgent, headers, 0);
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] postByte(String url) {
        return postByte(url, "");
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] postByte(String url, String parameter) {
        return postByte(url, parameter, "");
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] postByte(String url, String parameter, String userAgent) {
        return postByte(url, parameter, userAgent, null);
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] postByte(String url, String parameter, String userAgent, Map<String, String> headers) {
        return postByte(url, parameter, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送POST请求
     * @param url 发送请求的URL
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] postByte(String url, String parameter, String userAgent, Map<String, String> headers, int timeout) {
        return requestByte(url, Method.POST, parameter, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] requestByte(String url) {
        return requestByte(url, Method.GET, "");
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] requestByte(String url, Method method, String parameter) {
        return requestByte(url, method, parameter, "");
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] requestByte(String url, Method method, String parameter, String userAgent) {
        return requestByte(url, method, parameter, userAgent, null);
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] requestByte(String url, Method method, String parameter, String userAgent, Map<String, String> headers) {
        return requestByte(url, method, parameter, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送请求
     * @param url 发送请求的URL
     * @param method 请求方法
     * @param parameter 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static byte[] requestByte(String url, Method method, String parameter, String userAgent, Map<String, String> headers, int timeout) {
        byte[] result = null;
        OutputStreamWriter out = null;
        method = method != Method.POST ? Method.GET : Method.POST;
        try {
            URL realUrl = new URL(url + (method == Method.GET ? (url.indexOf("?") > -1 ? "&" : "?") + parameter : ""));
            URLConnection connection = realUrl.openConnection();

            // 设置超时时间
            connection.setConnectTimeout(timeout);
            connection.setReadTimeout(timeout);

            // 设置通用的请求属性
            userAgent = StringUtils.isNull(userAgent, "").trim();
            if (headers == null) {
                headers = new HashMap<>();
                headers.put("accept", "*/*");
                headers.put("connection", "Keep-Alive");
                headers.put("user-agent", userAgent.equals("") ? GENERAL_AGENT : userAgent);
            } else if (!userAgent.equals("")) {
                headers.put("user-agent", userAgent);
            }
            headers.entrySet().forEach((entry) -> {
                connection.setRequestProperty(entry.getKey(), entry.getValue());
            });

            // 设置请求配置
            if (method == Method.POST) {
                connection.setDoOutput(true);
                connection.setDoInput(true);

                out = new OutputStreamWriter(connection.getOutputStream(), "UTF-8");
                out.write(parameter);
                out.flush();
            } else {
                connection.connect();
            }

            // 读取URL的响应
            result = getBytes(connection.getInputStream());
        } catch (IOException e) {
            Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "发送" + method.toString() + "请求出现异常", e);
        } finally {
            try {
                if (out != null) { out.close(); }
            } catch (IOException e) {
                Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "关闭" + method.toString() + "请求出现异常", e);
            }
        }
        return result;
    }


    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file) {
        return request(url, file, "");
    }

    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @param data 请求参数
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file, Map<String, String> data) {
        return request(url, file, data, "");
    }

    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file, String userAgent) {
        return request(url, file, userAgent, null);
    }

    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @param data 请求参数
     * @param userAgent User-Agent
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file, Map<String, String> data, String userAgent) {
        return request(url, file, data, userAgent, null);
    }

    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file, String userAgent, Map<String, String> headers) {
        return request(url, file, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @param data 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file, Map<String, String> data, String userAgent, Map<String, String> headers) {
        return request(url, file, data, userAgent, headers, 0);
    }

    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file, String userAgent, Map<String, String> headers, int timeout) {
        return request(url, file, null, userAgent, headers, timeout);
    }

    /**
     * 向指定URL发送文件请求
     * @param url 发送请求的URL
     * @param file 文件
     * @param data 请求参数
     * @param userAgent User-Agent
     * @param headers 请求属性
     * @param timeout 超时时间（默认为0）
     * @return URL所代表远程资源的响应结果
     */
    public static String request(String url, File file, Map<String, String> data, String userAgent, Map<String, String> headers, int timeout) {
        String result = "";
        OutputStream out = null;
        DataInputStream in = null;
        BufferedReader reader = null;
        try {
            URL realUrl = new URL(url);
            URLConnection connection = realUrl.openConnection();

            // 设置超时时间
            connection.setConnectTimeout(timeout);
            connection.setReadTimeout(timeout);

            // 设置通用的请求属性
            userAgent = StringUtils.isNull(userAgent, "").trim();
            if (headers == null) {
                headers = new HashMap<>();
                headers.put("accept", "*/*");
                headers.put("charset", "UTF-8");
                headers.put("connection", "Keep-Alive");
                headers.put("user-agent", userAgent.equals("") ? GENERAL_AGENT : userAgent);
            } else if (!userAgent.equals("")) {
                headers.put("user-agent", userAgent);
            }

            // 设置边界
            String BOUNDARY = "----------" + System.currentTimeMillis();
            headers.put("Content-Type", "multipart/form-data; boundary=" + BOUNDARY);

            headers.entrySet().forEach((entry) -> {
                connection.setRequestProperty(entry.getKey(), entry.getValue());
            });

            // 设置请求配置
            connection.setDoInput(true);
            connection.setDoOutput(true);
            connection.setUseCaches(false);

            // 请求正文信息
            StringBuilder sb = new StringBuilder();
            sb.append("--"); // 必须多两道线
            sb.append(BOUNDARY);
            sb.append("\r\n");
            String regex = ".*/([^\\.]+)";
            sb.append(String.format("Content-Disposition: form-data;name=\"media\";filelength=\"" + file.length() + "\";filename=\"%s\"\r\n", file.getName().replaceAll(regex, "$1")));
            sb.append("Content-Type:application/octet-stream\r\n\r\n");
            byte[] head = sb.toString().getBytes("utf-8");

            // 获得输出流
            out = new DataOutputStream(connection.getOutputStream());
            out.write(head);

            // 把文件已流文件的方式 推入到url中
            in = new DataInputStream(new FileInputStream(file));
            int bytes = 0;
            byte[] bufferOut = new byte[1024];
            while ((bytes = in.read(bufferOut)) != -1) {
                out.write(bufferOut, 0, bytes);
            }
            in.close();

            // 写入其他数据
            if (data != null && !data.isEmpty()) {
                for (Map.Entry<String, String> entry : data.entrySet()) {
                    sb = new StringBuilder();
                    sb.append("\r\n--" + BOUNDARY + "\r\n");
                    sb.append("Content-Disposition: form-data;name=\"" + entry.getKey() + "\"\r\n\r\n");
                    sb.append(entry.getValue());
                    out.write(sb.toString().getBytes("utf-8"));
                }
            }

            // 结尾部分
            byte[] foot = ("\r\n--" + BOUNDARY + "--\r\n").getBytes("utf-8");// 定义最后数据分隔线
            out.write(foot);
            out.flush();
            out.close();

            // 读取URL的响应
            result = getText(connection.getInputStream());
        } catch (IOException e) {
            Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "发送文件请求出现异常", e);
        } finally {
            try {
                if (reader != null) { reader.close(); }
            } catch (IOException e) {
                Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "关闭文件请求出现异常", e);
            }
            try {
                if (in != null) { in.close(); }
            } catch (IOException e) {
                Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "关闭文件请求出现异常", e);
            }
            try {
                if (out != null) { out.close(); }
            } catch (IOException e) {
                Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "关闭文件请求出现异常", e);
            }
        }
        return result;
    }


    /**
     * 数据流转换文本
     * @param bytes 数据流
     * @return 文本
     */
    public static String getText(byte[] bytes) {
        try {
            return new String(bytes, "UTF-8");
        } catch (IOException e) {
            Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "数据流转换文本失败", e);
        }
        try {
            return new String(bytes);
        } catch (Exception e) {
            Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "数据流转换文本失败", e);
        }
        return "";
    }

    /**
     * 数据流转换文本
     * @param stram 数据流
     * @return 文本
     */
    public static String getText(InputStream stram) {
        String result = "";
        String line = null;
        InputStreamReader in = null;
        BufferedReader reader = null;

        try {
            in = new InputStreamReader(stram, "UTF-8");
            reader = new BufferedReader(in);
            while ((line = reader.readLine()) != null) { result += line; }
        } catch (Exception e) {
            Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "数据流转换文本失败", e);
        } finally {
            try {
                if (reader != null) { reader.close(); }
            } catch (IOException e) {
                Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "数据流转换文本失败", e);
            }
            try {
                if (in != null) { in.close(); }
            } catch (IOException e) {
                Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "数据流转换文本失败", e);
            }
        }

        return result;
    }

    /**
     * 数据流转换字节
     * @param stream 数据流
     * @return 字节
     */
    public static byte[] getBytes(InputStream stream) {
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        try {
            int read;
            byte[] bytes = new byte[1024];
            while ((read = stream.read(bytes, 0, 1024)) > 0) { out.write(bytes, 0, read); }
            return out.toByteArray();
        } catch (IOException e) {
            Logger.getLogger(WebUtils.class.getName()).log(Level.SEVERE, "数据流转换字节失败", e);
        }
        return null;
    }
}
